#!/bin/bash

rm -rf CenterNet_bs1.om CenterNet_bs16.om
source env.sh
atc --framework=5 --model=CenterNet.onnx  --output=CenterNet_bs1 --input_format=NCHW --input_shape="actual_input:1,3,512,512" --out_nodes="Conv_949:0;Conv_952:0;Conv_955:0" --log=info --soc_version=Ascend310
atc --framework=5 --model=CenterNet.onnx  --output=CenterNet_bs16 --input_format=NCHW --input_shape="actual_input:16,3,512,512" --out_nodes="Conv_949:0;Conv_952:0;Conv_955:0" --log=info --soc_version=Ascend310
#atc --input_format=NCHW --framework=5 --model=centernet2_1_bs1.onnx --input_shape="actual_input:1,3,512,512" --output=centernet2_1_bs1_auto_tune_addex --auto_tune_mode="RL,GA" --log=info --soc_version=Ascend310 --op_select_implmode=high_performance

if [ -f "CenterNet_bs1.om" ] && [ -f "CenterNet_bs16.om" ]; then
    echo "success"
else
    echo "fail!"
fi